"""
管理员功能页面
"""
from core import src
from interface import admin_interface

# 添加用户功能
def add_user():
    src.register()

# 修改额度功能
def change_balance():
    while True:
        # 输入要修改的用户名
        change_user = input('请输入您要修改的用户名：').strip()

        if change_user == 'q':
            break

        # 要修改的金额
        balance = input('请输入您要修改的额度：').strip()

        if not balance.isdigit():
            print('请输入数字！！')
            continue

        flag,msg = admin_interface.change_balance_interface(
            change_user,balance
        )
        if flag:
            print(msg)
            break

        else:
            print(msg)


# 冻结用户功能
def lock_user():
    while True:
        locked_user = input('请输入您要冻结的用户名：').strip()
        flag,msg = admin_interface.lock_user_interface(
            locked_user
        )

        if flag:
            print(msg)
            break
        else:
            print(msg)



# 管理员功能字典
admin_dic = {
    '1': add_user,
    '2': change_balance,
    '3': lock_user,
}



# 管理员启动函数
def admin_run():
    while True:
        print('''
        ============欢迎来到管理员页面============
                        1、添加用户
                        2、修改额度
                        3、冻结用户
        =================end===================
        ''')

        choice = input('请输入您选择的功能编号(温馨提示，输入q退出)：').strip()

        if choice == 'q':
            break

        if choice not in admin_dic:
            print('请输入正确的编号！！')
            continue

        admin_dic.get(choice)()

